python mock tests